package practice;

// 编写代码, 实现定时器案例

// 定时器还是比较那难以实现的！
// 定时器的主要方法就是 schedule
// 模拟实现【定时器】：
// 1. 创建MyTask类，存储参数 （任务描述+ 执行时间）——实现优先级就要比较
// 2. 创建MyTimer类：实现schedule方法

import java.util.concurrent.BlockingQueue;
import java.util.concurrent.PriorityBlockingQueue;

// 任务+重写比较规则
class MyTask implements Comparable<MyTask> {
    private Runnable runnable;
    private long time;

    public MyTask(Runnable runnable,long delay) {
        // 也就是任务的具体实现逻辑
        this.runnable = runnable;
        this.time = System.currentTimeMillis() + delay;  // 实际执行时间
    }

    // 获取
    public Runnable getRunnable() {
        return runnable;
    }

    public long getTime() {
        return time;
    }


    @Override
    public int compareTo(MyTask o) {
        // 按时间小->大
        return (int) (this.time - o.time);
    }
}

// MyTimer实现schedule方法
class MyTimer {
    // 创建一个对象
    private Object locker = new Object();

    // 任务存入优先阻塞队列中
    private BlockingQueue<MyTask> queue = new PriorityBlockingQueue<>();

    public void schedule(Runnable runnable, long delay) throws InterruptedException {
        MyTask task = new MyTask(runnable,delay);
        // 任务加入到阻塞队列中
        queue.put(task);

        // 任务加入后就可以唤醒任务的执行
        // 加锁 原子操作
        synchronized (locker) {
            locker.notify();
        }
    }

    public MyTimer() {
        // 创建线程 执行任务
        Thread t = new Thread(() -> {
            long curTime = System.currentTimeMillis();

            while (true) {
                try {
                    // 保证原子性
                    synchronized (locker) {
                        // 取出队首元素，然后再看是否可以直接执行任务
                        MyTask task = queue.take();
                        // 进行比较是否执行任务
                        // 注意比较的是当前刚获取的时间 而不是上面之前的curTime
                        if(task.getTime() <= System.currentTimeMillis()) {
                            // 注意逻辑：当前时间超过设定时间就执行任务
                            task.getRunnable().run();
                        } else {
                            // 还没有到执行时间就进行阻塞
                            // 还要将这个任务再塞回去 等时间到了再执行
                            // 设置等待时间 避免忙等
                            queue.put(task);
                            locker.wait(task.getTime()-curTime);
                        }
                    }
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }

            }
        });
        t.start();
    }
}

public class Main2 {
    public static void main(String[] args) throws InterruptedException {
        MyTimer timer = new MyTimer();
        timer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("时间到喽！");
            }
        },6000);

        timer.schedule(new Runnable() { // 注意参数类型：Runnable的实例对象！！
            @Override
            public void run() {
                System.out.println("时间到ge！");
            }
        },600);

        System.out.println("开始计时！");
    }
}
